/******************************************************************************
 * Copyright (C) 2015  671643387@qq.com
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 *****************************************************************************/

#ifndef SHARED_AUTH_AUTHCRYPT_H_
#define SHARED_AUTH_AUTHCRYPT_H_

#if defined(_MSC_VER) && (_MSC_VER >= 1200)
# pragma once
#endif

#include <cstddef>
#include <cstdint>
#include <vector>
#include "Ucop/Macros.h"
#include "Ucop/Crypto/BigNumber.h"
using Ucop::Crypto::BigNumber;


class AuthCrypt
{
public:
	AuthCrypt(void);
	~AuthCrypt(void);

	const static std::size_t CRYPTED_SEND_LEN = 4;
	const static std::size_t CRYPTED_RECV_LEN = 6;

	void Init(void);
	void SetKey(std::uint8_t *_Key, std::size_t _Size);
	void DecryptRecv(std::uint8_t *_Data, std::size_t _Size);
	void EncryptSend(std::uint8_t *_Data, std::size_t _Size);
	inline bool IsInitialized(void) const {
		return _Myinited;
	}

	static void GenerateKey(std::uint8_t *_Key, BigNumber *_Bn);

private:
	std::vector<std::uint8_t> _Mykey;
	std::uint8_t _Mysend_i, _Mysend_j, _Myrecv_i, _Myrecv_j;
	bool _Myinited;
	DISALLOW_COPY_AND_ASSIGN(AuthCrypt);
};

#endif /* SHARED_AUTH_AUTHCRYPT_H_ */
